package com.huilan.etl.util;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.huilan.etl.entity.Head;
import com.taocares.sdxp.model.HeadType;
import com.taocares.sdxp.model.MsgType;
import com.taocares.sdxp.util.JAXBUtil;
import org.dom4j.*;
import org.springframework.util.StringUtils;

import java.util.List;

/**
 * @author ghy
 * @date 2018-12-28 09:32
 **/
public class XmlUtil {

    /**
     * xml服务编码
     */
    private static final String SVC_SERVICECODE="Msg/Head/Svc_ServiceCode";


    /**
     * xml根节点
     */
    private static final String MSG="Msg";

    /**
     * xml头部
     */
    private static final String HEAD="Head";

    /**
     * xml内容
     */
    private static final String BODY="Body";


    /**
     * 从xml中获取服务编码
     * @param xml xml字符串
     * @return 服务编码
     */
    public static String  getSvcServiceCode(String xml) throws Exception {
            MsgType msgType = JAXBUtil.unmarshalIgnoreNameSpace(MsgType.class, xml);
            return msgType.getHead().getSvcServiceCode();
//        Document doc = null;
//        try {
//            doc = DocumentHelper.parseText(xml);
//            Node node = doc.selectSingleNode(SVC_SERVICECODE);
//            return node.getText();
//        } catch (DocumentException e) {
//            e.printStackTrace();
//        }
    }

    /**
     * xml字符串转json对象
     * @param xml xml字符串
     * @return json对象
     */
    private static JSONObject xml2json(String xml){
        JSONObject jsonObject = new JSONObject();
        Document document = null;
        try {
            document = DocumentHelper.parseText(xml);
            //获取根节点元素对象
            Element root = document.getRootElement();
            iterateNodes(root, jsonObject);
            return jsonObject;
        } catch (DocumentException e) {
            e.printStackTrace();
        }
        return jsonObject;
    }
    /**
     * 遍历元素
     * @param node 元素
     * @param json 将元素遍历完成之后放的JSON对象
     */
    @SuppressWarnings("unchecked")
    private static void iterateNodes(Element node, JSONObject json){
        //获取当前元素的名称
        String nodeName = node.getName();
        //判断已遍历的JSON中是否已经有了该元素的名称
        if(json.containsKey(nodeName)){
            //该元素在同级下有多个
            Object object = json.get(nodeName);
            JSONArray array = null;
            if(object instanceof JSONArray){
                array = (JSONArray) object;
            }else {
                array = new JSONArray();
                array.add(object);
            }
            //获取该元素下所有子元素
            List<Element> listElement = node.elements();
            if(listElement.isEmpty()){
                //该元素无子元素，获取元素的值
                String nodeValue = node.getTextTrim();
                array.add(nodeValue);
                json.put(nodeName, array);
                return ;
            }
            //有子元素
            JSONObject newJson = new JSONObject();
            //遍历所有子元素
            for(Element e:listElement){
                //递归
                iterateNodes(e,newJson);
            }
            array.add(newJson);
            json.put(nodeName, array);
            return ;
        }
        //该元素同级下第一次遍历
        //获取该元素下所有子元素
        List<Element> listElement = node.elements();
        if(listElement.isEmpty()){
            //该元素无子元素，获取元素的值
            String nodeValue = node.getTextTrim();
            json.put(nodeName, nodeValue);
            return ;
        }
        //有子节点，新建一个JSONObject来存储该节点下子节点的值
        JSONObject object = new JSONObject();
        //遍历所有一级子节点
        for(Element e:listElement){
            //递归
            iterateNodes(e,object);
        }
        json.put(nodeName, object);
    }


    /**
     * 获取Msg节点信息
     */
    public static MsgType getMsg(String xml) throws Exception {
        return JAXBUtil.unmarshalIgnoreNameSpace(MsgType.class, xml);
//        return xml2json(xml).getJSONObject(XmlUtil.MSG);
    }

    /**
     * 获取HEAD节点Bean信息
     */
    public static HeadType getHeadBean(String xml) throws Exception {
        MsgType msgType = JAXBUtil.unmarshalIgnoreNameSpace(MsgType.class, xml);
        return msgType.getHead();
//        JSONObject jsonObject = xml2json(xml).getJSONObject(XmlUtil.MSG).getJSONObject(XmlUtil.HEAD);
//        return JSONObject.toJavaObject(jsonObject, Head.class);
    }

    /**
     * 获取BODY节点信息
     */
    public static JSONObject getBodyJson(String xml) {
        return xml2json(xml).getJSONObject(XmlUtil.MSG).getJSONObject(XmlUtil.BODY);
    }



    /**
     * 获取BODY内实体信息
     */
    public static <T> T getBodyBean(String xml,Class<T> clazz){
        JSONObject jsonObject = xml2json(xml).getJSONObject(XmlUtil.MSG).getJSONObject(XmlUtil.BODY).getJSONObject(clazz.getSimpleName());
        return JSONObject.toJavaObject(jsonObject,clazz);
    }


    /**
     * 获取BODY内实体信息
     */
    public static <T> T getBodyBean(String xml,Class<T> clazz,String className){

        if(StringUtils.isEmpty(className)){
            className = clazz.getSimpleName();
        }

        JSONObject jsonObject = xml2json(xml).getJSONObject(XmlUtil.MSG).getJSONObject(XmlUtil.BODY).getJSONObject(className);
        return JSONObject.toJavaObject(jsonObject,clazz);
    }

    /**
     * 获取BODY内实体集合信息
     * @param xml xml字符串
     * @param clazz 需转换的JavaBean
     * @param <T> 返回的泛型
     * @return 返回
     */
    public static <T> List<T> getBodyBeanList(String xml,Class<T> clazz,String className){
        return getBeanList(xml,null,clazz,className);
    }


    /**
     * 获取BODY内实体集合信息
     * @param xml xml字符串
     * @param clazz 需转换的JavaBean
     * @param <T> 返回的泛型
     * @return 返回
     */
    public static <T> List<T> getBodyBeanList(String xml,Class<T> clazz,String className,String classList){
        return getBeanList(xml,null,clazz,className);
    }


    /**
     * 获取BODY内实体集合信息
     * @param xml xml字符串
     * @param classList 需解析的集合名称
     * @param clazz 需转换的JavaBean
     * @param <T> 返回的泛型
     * @return 返回
     */
    public static <T> List<T> getBodyBeanList(String xml,String classList,Class<T> clazz,String className){

        return getBeanList(xml,classList,clazz,className);
    }


    /**
     * 获取xml body内对象集合
     * @param xml xml字符串
     * @param classList 需解析的集合名称
     * @param clazz 需转换的JavaBean
     * @param <T> 返回的泛型
     * @return 返回
     */
    private static <T> List<T> getBeanList(String xml,String classList,Class<T> clazz,String className) {

        if(StringUtils.isEmpty(className)){
            className = clazz.getSimpleName();
        }

        if(StringUtils.isEmpty(classList)){
            classList = className+"List";
        }

        JSONObject jsonObject = xml2json(xml).getJSONObject(XmlUtil.MSG).getJSONObject(XmlUtil.BODY).getJSONObject(classList);
        Object object = jsonObject.get(className);

        //如果object是json数组
        if(object instanceof JSONArray){
            JSONArray jsonArray = jsonObject.getJSONArray(className);
            return JSONObject.parseArray(jsonArray.toJSONString(),clazz);
            //如果object是json对象
        }else if(object instanceof JSONObject){
            JSONObject json = jsonObject.getJSONObject(className);
            JSONArray jsonArray = new JSONArray();
            jsonArray.add(json);
            return JSONObject.parseArray(jsonArray.toJSONString(),clazz);
        }else{
            return null;
        }
    }


}
